from __future__ import annotations

from pydantic import Field

from app.schemas.common import APIModel


class AISuggestionRequest(APIModel):
  entry_context: str
  mood: str | None = None
  tags: list[str] = Field(default_factory=list)
  provider: str | None = None


class AISuggestionResponse(APIModel):
  message: str
  tags: list[str] = Field(default_factory=list)
  prompt_id: str | None = None


class AIChatRequest(APIModel):
  thread_id: str | None = None
  message: str
  provider: str | None = None


class AIChatResponse(APIModel):
  thread_id: str
  reply: str
  relationship_id: int | None = None


class AIAssistantRequest(APIModel):
  message: str
  provider: str | None = None


class AIAssistantResponse(APIModel):
  reply: str
